package com.wchen.project.zk.demo;

import org.apache.zookeeper.*;

import java.util.List;
import java.util.concurrent.CountDownLatch;

/**
 * @Desc:   zookeeper API 获取子节点列表，使用同步 sync 接口
 * @author: cww
 * @DateTime: 2020/1/17 11:09
 */

public class ZooKeeperGetChildrenSyncAPI implements Watcher {
    private static CountDownLatch connectedSemaphore = new CountDownLatch(1);
    private static ZooKeeper zk = null;

    public static void main(String[] args) throws Exception{
        String path = "/zk-book";
        zk = new ZooKeeper("home:2181", 5000, new ZooKeeperGetChildrenSyncAPI());
        connectedSemaphore.await();
//        zk.delete(path, 0);
        zk.create(path, "".getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
        zk.create(path + "/c1", "".getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);

        List<String> childrenList = zk.getChildren(path, true);
        System.out.println("first getChildren: " + childrenList);

        zk.create(path + "/c2", "".getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);
        zk.create(path + "/c3", "".getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);

        Thread.sleep(5000);
    }

    /**
     * getChildren API
     * 参数：
     * @Watcher watcher  注册的 Watcher，在获取了本次子节点之后，如果这个子节点列表发生变更，那么就会向客户端发送通知，允许null
     * @boolean watch    是否要注册一个 Watcher，这里的 Watcher 是默认 Watcher，如果是true ，那么客户端就会用默认 Watcher
     *                  进行注册事件监听
     * @ChildrenCallBack cb  注册一个异步回调函数
     * @Object ctx       用于传递上下文信息对象
     * @Stat stat        指定数据节点的节点状态信息，在接口中传入一个旧的 stat 变量，在 stat 变量会在方法执行过程中，被来自服务端
     *                  新的 stat 对象替换
     * @param event
     */
    @Override
    public void process(WatchedEvent event) {
        if (Event.KeeperState.SyncConnected == event.getState()) {
            if (Event.EventType.None == event.getType() && null == event.getPath()) {
                connectedSemaphore.countDown();
            } else if (event.getType() == Event.EventType.NodeChildrenChanged) {
                // 服务端触发了 Children changed 事件仅仅会发出一个通知，而不会吧节点的变化情况发送给客户端，需要如下
                // 代码所示客户端自己重新获取
                try {
                    // true 表示注册一个默认 Watcher，如果下次要继续监控，则一直要为true，因为触发一次通知后就失效了
                    System.out.println("ReGet Child: " + zk.getChildren(event.getPath(), true));
                } catch (KeeperException e) {
                    e.printStackTrace();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}
